from os import walk
import pygame

def import_folder(path):
    # 导入文件夹中的图像文件。
    # 参数：
    # - path：要导入的文件夹的路径。
    # 返回：
    # 包含图像表面对象的列表。
    surface_list = []

    for _, __, img_files in walk(path):
        for image in img_files:
            full_path = path + '/' + image
            image_surf = pygame.image.load(full_path).convert_alpha()
            surface_list.append(image_surf)

    return surface_list

def import_folder_dict(path):
    # 导入文件夹中的图像文件，并将文件名作为键名构建字典。
    # 参数：
    # - path：要导入的文件夹的路径。
    # 返回：
    # 包含图像表面对象的字典，其中键是文件名（不包含扩展名）。
    surface_dict = {}

    for _, __, img_files in walk(path):
        for image in img_files:
            full_path = path + '/' + image
            image_surf = pygame.image.load(full_path).convert_alpha()
            surface_dict[image.split('.')[0]] = image_surf

    return surface_dict
